import { createParamDecorator } from '@nestjs/common'

/**
 * 自定义一个路由处理函数的参数装饰器
 * 传递一个工厂函数 用于获取值
 */
export const User = createParamDecorator((data, ctx) => {
  // 获取请求对象
  const req = ctx.switchToHttp().getRequest()
  return data ? req.user[data] : req.user
})